import sys
sys.path.append('..')
from datetime import datetime
from models import db, Holiday

holidays = {
    '2018-12-30': '元旦',
    '2018-12-31': '元旦',
    '2019-01-01': '元旦',
    '2019-02-04': '春节',
    '2019-02-05': '春节',
    '2019-02-06': '春节',
    '2019-02-07': '春节',
    '2019-02-08': '春节',
    '2019-02-09': '春节',
    '2019-02-10': '春节',
    '2019-04-05': '清明节',
    '2019-04-06': '清明节',
    '2019-04-07': '清明节',
    '2019-05-01': '劳动节',
    '2019-06-07': '端午节',
    '2019-06-08': '端午节',
    '2019-06-09': '端午节',
    '2019-09-13': '中秋节',
    '2019-09-14': '中秋节',
    '2019-09-15': '中秋节',
    '2019-10-01': '国庆',
    '2019-10-02': '国庆',
    '2019-10-03': '国庆',
    '2019-10-04': '国庆',
    '2019-10-05': '国庆',
    '2019-10-06': '国庆',
    '2019-10-07': '国庆'
};


holidaysOffset = {
    '2018-12-29': '元旦补班',
    '2019-02-02': '春节补班',
    '2019-02-03': '春节补班',
    '2019-09-29': '国庆日补班',
    '2019-10-12': '国庆日补班'
};


for k,v in holidays.items():
    try:
        date = datetime.strptime(k, '%Y-%m-%d')
        db.session.add(Holiday(date=date,desc=v))
        db.session.commit()
        print(f'Add {k} Successfully!')
    except:
        db.session.rollback()


for k,v in holidaysOffset.items():
    try:
        date = datetime.strptime(k, '%Y-%m-%d')
        db.session.add(Holiday(date=date,desc=v,isworkday=True))
        db.session.commit()
        print(f'Add {k} Successfully!')
    except:
        db.session.rollback()


